Tensorflow error: Failed to serialize message. For multi-modal dataset